package cn.nawang.ebeim.client.db.operate;

import java.io.File;
import java.io.IOException;
import java.io.OutputStream;
import java.io.OutputStreamWriter;

import cn.nawang.ebeim.client.bean.DataSource;
import cn.nawang.ebeim.client.util.Constants;
import cn.nawang.ebeim.client.util.FileUtil;
import cn.nawang.ebeim.client.util.PropertiesUtil;
import cn.nawang.ebeim.client.util.SystemUtil;

/**
 * 在进行导出的时候，需要注意命令语句的运行环境，如果已经将mysql安装路径下的bin加入到 系统的path变量中，那么在导出的时候可以直接使用命令语句，否则，就需要在执行命令语句的
 * 时候加上命令所在位置的路径，即mysql安装路径想的bin下的mysqldump命令
 * 
 * @ClassName: MySqlImportAndExport
 * @Description: MYSQL数据库导入导出工具类
 * @author 张建峰
 * @date 2015年9月6日 下午5:54:08
 * @version v1.0
 */
public class MySqlImportAndExport implements DbImportAndExport{

  public static void main(String args[]) {
    long s = System.currentTimeMillis();
    DataSource ds = new DataSource();
    ds.setId("fasdfasdfa");
    ds.setDbHost("127.0.0.1");
    ds.setDbName("ebeim");
    ds.setDbPassword("root");
    ds.setDbUsername("root");
    ds.setDbPort("3306");
    ds.setDbType("0");
    try {
      new MySqlImportAndExport(ds).exportDb();
    } catch (Exception e1) {
      e1.printStackTrace();
    }
    long e = System.currentTimeMillis();
    System.out.println(e - s);
  }
  
  private DataSource ds;
  
  public MySqlImportAndExport(DataSource ds){
    this.ds = ds;
  }

  /**
   * 根据属性文件的配置导出指定位置的指定数据库到指定位置
   * 
   * @param properties
   * @throws IOException
   */
  public int exportDb() throws Exception {
    Runtime runtime = Runtime.getRuntime();
    String command = getExportCommand();
    Process proc = runtime.exec(command);
    int exitVal = proc.waitFor();  
    if(exitVal == 0){
      System.out.println("数据库导出成功！");
    }else{
      System.out.println("数据库导出失败，可能原因：数据库用户名或密码不正确");
    }
    return exitVal;
  }
  
  /**
   * 利用属性文件提供的配置来拼装命令语句 在拼装命令语句的时候有一点是需要注意的：一般我们在命令窗口直接使用命令来 进行导出的时候可以简单使用“>”来表示导出到什么地方，即mysqldump
   * -uusername -ppassword databaseName > exportPath， 但在Java中这样写是不行的，它需要你用-r明确的指出导出到什么地方，如：
   * mysqldump -uusername -ppassword databaseName -r exportPath。
   * 
   * @param properties
   * @return
   * @throws Exception 
   */
  private String getExportCommand() throws Exception {
    
    StringBuffer command = new StringBuffer();
    String username = ds.getDbUsername();// 用户名
    String password = ds.getDbPassword();// 用户密码
    String exportDatabaseName = ds.getDbName();// 需要导出的数据库名
    String host = ds.getDbHost();// 从哪个主机导出数据库，如果没有指定这个值，则默认取localhost
    String port = ds.getDbPort();// 使用的端口号
    String exportPath = FileUtil.getDbUploadPath(ds.getId(),ds.getDbName()) + Constants.DB_SUFFIX;
    File f = new File(exportPath);
    if(!f.getParentFile().exists()){
      f.getParentFile().mkdirs();
    }

    // 注意哪些地方要空格，哪些不要空格
    command.append(SystemUtil.getBinPath(new PropertiesUtil())+File.separator)
        .append("mysqldump -u").append(username)
        .append(" -p").append(password)// 密码是用的小p，而端口是用的大P。
        .append(" -h").append(host).append(" -P").append(port).append(" ")
        .append(exportDatabaseName).append(" -r ").append(exportPath);
    return command.toString();
  }

  /**
   * 根据属性文件的配置把指定位置的指定文件内容导入到指定的数据库中 在命令窗口进行mysql的数据库导入一般分三步走： 第一步是登到到mysql； mysql -uusername
   * -ppassword -hhost -Pport -DdatabaseName;如果在登录的时候指定了数据库名则会 直接转向该数据库，这样就可以跳过第二步，直接第三步；
   * 第二步是切换到导入的目标数据库；use importDatabaseName； 第三步是开始从目标文件导入数据到目标数据库；source importPath；
   * 
   * @param properties
   * @throws Exception 
   * @throws IOException
   */
  public void importDb() throws Exception {
    Runtime runtime = Runtime.getRuntime();
    // 因为在命令窗口进行mysql数据库的导入一般分三步走，所以所执行的命令将以字符串数组的形式出现
    try {
      String cmdarray[] = getImportCommand();// 根据属性文件的配置获取数据库导入所需的命令，组成一个数组
      // runtime.exec(cmdarray);//这里也是简单的直接抛出异常
      Process process = null;
      process = runtime.exec(cmdarray[0]);
      // 执行了第一条命令以后已经登录到mysql了，所以之后就是利用mysql的命令窗口
      // 进程执行后面的代码
      OutputStream os = process.getOutputStream();
      OutputStreamWriter writer = new OutputStreamWriter(os);
      // 命令1和命令2要放在一起执行
      writer.write(cmdarray[1] + "\r\n" + cmdarray[2]);
      writer.flush();
      writer.close();
      os.close();
      
      int exitVal = process.waitFor();  
      if(exitVal == 0){
        System.out.println("操作完成！");
      }else{
        System.out.println("操作失败，可能原因：数据库用户名或密码不正确");
      }
      
    } catch (IOException e) {
      System.out.println("操作失败，可能原因：(1)未找到数据库安装路径下的bin目录,(2)目标文件找不到");
      e.printStackTrace();
    } catch (InterruptedException e) {
      e.printStackTrace();
    }
  }

  /**
   * 根据属性文件的配置，分三步走获取从目标文件导入数据到目标数据库所需的命令 如果在登录的时候指定了数据库名则会 直接转向该数据库，这样就可以跳过第二步，直接第三步；
   * 
   * @param properties
   * @return
   * @throws Exception 
   */
  private String[] getImportCommand() throws Exception {
    String username = ds.getDbUsername();// 用户名
    String password = ds.getDbPassword();// 密码
    String host = ds.getDbHost();// 导入的目标数据库所在的主机
    String port = ds.getDbPort();// 使用的端口号
    String importDatabaseName = ds.getDbName();// 导入的目标数据库的名称
    String importPath = FileUtil.getDbDownloadPath(ds.getId(),ds.getDbName());// 导入的目标文件所在的位置
    File f = new File(importPath);
    if(!f.getParentFile().exists()){
      f.getParentFile().mkdirs();
    }
    // 第一步，获取登录命令语句
    String loginCommand =
        new StringBuffer()
            .append(SystemUtil.getBinPath(new PropertiesUtil())+File.separator)
            .append("mysql -u")
            .append(username).append(" -p").append(password).append(" -h").append(host)
            .append(" -P").append(port).toString();
    // 第二步，获取切换数据库到目标数据库的命令语句
    String switchCommand = new StringBuffer("use ").append(importDatabaseName).toString();
    // 第三步，获取导入的命令语句
    String importCommand = new StringBuffer("source ").append(importPath).toString();
    // 需要返回的命令语句数组
    String[] commands = new String[] {loginCommand, switchCommand, importCommand};
    return commands;
  }

}
